from os import system as s
import json


s('pip install kaggle')
s('pip install -i https://pypi.flyai.com/simple flyai==0.6.6')
s('pip install tensorboard==1.14.0')
s('pip install future')

s('mkdir .kaggle')

token = {"username": "zcwjack", "key": "c16800f9df3c594056fc92ee4f2da794"}
with open('/root/.kaggle/kaggle.json', 'w') as file:
    json.dump(token, file)

s('kaggle config set -n path -v{/root}')
s('chmod 600 /root/.kaggle/kaggle.json')
s('kaggle datasets download -d zcwjack/baldclassificationflyai -p /root')
s('kaggle datasets download -d zcwjack/model-baldclassification -p /root')
s('unzip \*.zip')
# s('cp -r BaldModel/* .')
s('cp -r Desktop/* .')
s('cp -r data ./BaldClassification/')
# s('mkdir ./data/output')
# s('cp resnest50_bald.pt ./data/output/model/model11.pt')
# s('cp resnest50_bald.pt ./data/output/model/model12.pt')
# s('cp densenet121_bald.pt ./data/output/model/model21.pt')
# s('cp densenet121_bald.pt ./data/output/model/model22.pt')
